import unittest
# 1.导入参数化插件
from parameterized import parameterized
# 通过一个函数提供测试数据
def get_data():
    # 获取了一组测试数据，最后返回出去
    return [(1, 2, 3), (3, 4, 7), (5, 6, 11), (7, 8, 15)]
def add(a, b):
    return a + b
class TestAdd(unittest.TestCase):
    # 测试加法
    # 2.为测试方法提供所需的参数
    # @parameterized.expand([
    #     (1, 2, 3),
    #     (3, 4, 7),
    #     (5, 6, 11),
    #     (7, 8, 16),
    # ])
    @parameterized.expand(get_data)
    def test_add(self, n1, n2, expect):
        # 调用函数
        result = add(n1, n2)
        # print('运行结果：', result)
        # 断言: 运行结果与预期结果对比
        self.assertEqual(result, expect)
if __name__ == '__main__':
    unittest.main()

